package com.ecommerce.gateway.util;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.security.Keys;
import org.springframework.stereotype.Component;

import javax.crypto.SecretKey;
import java.util.Date;

/**
 * JWT工具类
 */
@Component
public class JwtUtil {
    
    // 使用安全的密钥生成方式，与认证服务保持一致
    private static final SecretKey SECRET_KEY = Keys.hmacShaKeyFor("shopping-microservices-secret-key-2023".getBytes());

    /**
     * 从JWT令牌中提取用户名
     * @param token JWT令牌
     * @return 用户名
     */
    public String extractUsername(String token) {
        Claims claims = Jwts.parser()
                .verifyWith(SECRET_KEY)
                .build()
                .parseSignedClaims(token)
                .getPayload();
        return claims.getSubject();
    }

    /**
     * 从JWT令牌中提取用户ID（字符串类型）
     * @param token JWT令牌
     * @return 用户ID（字符串类型）
     */
    public String extractUserIdAsString(String token) {
        Claims claims = Jwts.parser()
                .verifyWith(SECRET_KEY)
                .build()
                .parseSignedClaims(token)
                .getPayload();
        Object userIdObj = claims.get("userId");
        if (userIdObj == null) {
            return null;
        }
        // 统一将userId转换为String类型
        if (userIdObj instanceof Long) {
            return userIdObj.toString();
        } else if (userIdObj instanceof Integer) {
            return ((Integer) userIdObj).toString();
        } else if (userIdObj instanceof String) {
            String userIdStr = (String) userIdObj;
            return userIdStr != null ? userIdStr.trim() : null;
        }
        return null;
    }

    /**
     * 从JWT令牌中提取用户ID（Long类型）
     * @param token JWT令牌
     * @return 用户ID（Long类型）
     */
    public Long extractUserId(String token) {
        String userIdStr = extractUserIdAsString(token);
        if (userIdStr != null && !userIdStr.trim().isEmpty()) {
            try {
                return Long.parseLong(userIdStr.trim());
            } catch (NumberFormatException e) {
                return null;
            }
        }
        return null;
    }

    /**
     * 从JWT令牌中提取角色
     * @param token JWT令牌
     * @return 角色
     */
    public String extractRole(String token) {
        Claims claims = Jwts.parser()
                .verifyWith(SECRET_KEY)
                .build()
                .parseSignedClaims(token)
                .getPayload();
        return claims.get("role", String.class);
    }

    /**
     * 检查JWT令牌是否无效
     * @param token JWT令牌
     * @return 是否无效
     */
    public boolean isInvalid(String token) {
        try {
            // 验证签名
            Jwts.parser().verifyWith(SECRET_KEY).build().parseSignedClaims(token);
            
            // 检查是否过期
            Claims claims = Jwts.parser()
                    .verifyWith(SECRET_KEY)
                    .build()
                    .parseSignedClaims(token)
                    .getPayload();
            Date expiration = claims.getExpiration();
            return expiration.before(new Date());
        } catch (Exception e) {
            return true;
        }
    }
}